// file name: test.go

package conf

import (
	"encoding/json"
	"log"
	"os"
)

var confDir string

type configure struct {
	Engine string `json:"engine"`
}

func init() {
	homePath := os.Getenv("HOME")
	confDir = homePath + "/.translate"
}

func marshal(c configure) []byte {
	data, err := json.Marshal(c)
	if err != nil {
		log.Fatal(err)
	}
	return data
}

func unmarshal(data []byte) configure {
	var c configure
	err := json.Unmarshal(data, &c)
	if err != nil {
		log.Fatal(err)
	}

	return c
}

func read() []byte {
	_, err := os.Stat(confDir)
	if err != nil { //文件不存在
		fileInit()
	}

	fp, err := os.OpenFile(confDir, os.O_RDONLY, 0755)
	defer fp.Close()
	if err != nil {
		log.Fatal(err)
	}
	data := make([]byte, 100)
	n, err := fp.Read(data)
	if err != nil {
		log.Fatal(err)
	}
	return data[:n]
}

func write(data []byte) {

	err1 := os.Truncate(confDir, 0) //清空文件
	if err1 != nil {
		log.Fatal(err1)
	}

	fp, err := os.OpenFile(confDir, os.O_RDWR|os.O_CREATE, 0755)
	if err != nil {
		log.Fatal(err)
	}
	defer fp.Close()
	_, err = fp.Write(data)
	if err != nil {
		log.Fatal(err)
	}
}

// 文件初始化
func fileInit() {
	confData := configure{
		Engine: "youdao",
	}

	data := marshal(confData)
	write(data)
}

// 获取相关配置
func GetConf() configure {
	data := read()
	var c configure
	err := json.Unmarshal(data, &c)
	if err != nil {
		log.Fatal(err)
	}
	return c
}

// 设置引擎
func SetEngine(e string) {
	confData := configure{
		Engine: e,
	}
	data := marshal(confData)
	write(data)
}
